#pragma once

#include <iostream>
#include <functional>
#include <cstring>
#include <cstdio>
#include <pthread.h>

namespace quitesix
{
    static int number = 1;
    template <typename T>
    class Thread
    {
        using func_t = std::function<void(T)>;

    private:
        void EnableRunning()
        {
            _isrunning = true;
        }
        void EnableDetach()
        {
            _isdetach = true;
        }
        // 静态成员函数原因：非静态成员函数第一个参数是隐含的this指针，参数不匹配
        static void *Routine(void *args)
        {
            Thread<T> *self = static_cast<Thread<T> *>(args);
            self->EnableRunning();
            if (self->_isdetach)
                self->Detach();
            // 设置名字
            pthread_setname_np(self->_tid, self->_name.c_str());
            // 回调
            (self->_func)(self->_data);
            return nullptr;
        }

    public:
        Thread(func_t func, T data)
            : _func(func), _isdetach(false), _isrunning(false), _res(nullptr), _data(data)
        {
            _name = "thread-" + std::to_string(number++);
        }
        ~Thread()
        {
        }

        bool Start()
        {
            if (_isrunning)
                return false;
            // 传递this指针，因为静态成员函数不存在this指针，需要访问this指针中_func进行回调
            int n = pthread_create(&_tid, nullptr, Routine, this);
            if (n != 0)
            {
                std::cerr << "线程创建失败，" << strerror(n) << std::endl;
                return false;
            }
            std::cout << "线程创建成功！" << std::endl;
            if (_isdetach)
                Detach();
            return true;
        }
        void Stop()
        {
            if (_isrunning)
            {
                int n = pthread_cancel(_tid);
                if (n != 0)
                {
                    std::cerr << "线程取消失败，" << strerror(n) << std::endl;
                    return;
                }
                std::cout << "线程取消成功" << std::endl;
            }
        }
        void Detach()
        {
            // 已经分离了，直接退
            if (_isdetach)
                return;
            // 在运行且没有分离，分离
            if (_isrunning)
                pthread_detach(_tid);
            EnableDetach();
            std::cout << "线程分离成功" << std::endl;
        }
        void Join()
        {
            if (!_isrunning)
                return;
            int n = pthread_join(_tid, nullptr);
            if (n != 0)
            {
                std::cerr << "线程等待失败，" << strerror(n) << std::endl;
                return;
            }
            std::cout << "线程等待成功" << std::endl;
        }

    private:
        pthread_t _tid;    // 线程ID
        func_t _func;      // 线程要执行的事务
        bool _isdetach;    // 是否分离
        bool _isrunning;   // 线程是否运行
        std::string _name; // 线程名
        void *_res;        // 返回值
        T _data;           // 参数
    };
}